../tiny_llm/models.py